# 数据库初始化
from tortoise.contrib.fastapi import register_tortoise
from fastapi import FastAPI


def init_db(app: FastAPI):
    register_tortoise(
        app,
        db_url="sqlite://db.sqlite3",  # 可替换为 mysql/postgres
        modules={"models": ["models.user"]},  # 关键点
        generate_schemas=True,  # 启动时自动创建表
        add_exception_handlers=True
    )